import logging

from django.core.cache import cache
from django.http import JsonResponse
from django.shortcuts import render
from django.views import View

from apps.areas.models import Area
from utils.response_code import RETCODE

logger = logging.getLogger('django')  # 日志记录器

class AreasView(View):
    def get(self, request):
        # 获取请求参数area_id
        area_id = request.GET.get('area_id')
        # 判断是否提供了area_id参数
        if not area_id:
            province_list = cache.get('province_list')
            if not province_list:
                # 提供省份数据
                # 查询省份数据
                try:
                    province_model_list = Area.objects.filter(parent__isnull=True)
                    # 构造响应数据（省份数据）
                    province_list = []
                    for province_mode in province_model_list:
                        province_list.append({'id': province_mode.id, 'name': province_mode.name})
                    # 缓存省份数据
                    cache.set('province_list', province_list, 3600)

                except Exception as e:
                    logger.error(e)
                    return JsonResponse({'code': RETCODE.DBERR, 'errmsg': '省份数据错误'})
            # 返回响应数据
            return JsonResponse({'code': RETCODE.OK, 'errmsg': 'OK', 'province_list': province_list})
        # 提供市或区的数据
        else:
            # 获取缓存数据
            sub_data = cache.get('sub_data_' + area_id)
            if not sub_data:
                # 查询市或区的数据（先查询市或区的父级，然后在根据父级获取当前父级下所有的下属城市）
                try:
                    parent_model = Area.objects.get(id=area_id)
                    sub_model_list = parent_model.subs.all()
                    # 构造响应数据（市或区的数据）
                    sub_list = []
                    for sub_model in sub_model_list:
                        sub_list.append({'id': sub_model.id, 'name': sub_model.name})
                    sub_data = {
                        'id': parent_model.id,
                        'name': parent_model.name,
                        'subs': sub_list
                    }
                    # 缓存市或区数据
                    cache.set('sub_data_' + area_id, sub_data, 3600)
                except Exception as e:
                    logger.error(e)
                    return JsonResponse({'code': RETCODE.DBERR, 'errmsg': '城市或区数据错误'})
            # 返回响应数据
            return JsonResponse({'code': RETCODE.OK, 'errmsg': 'OK', 'sub_data': sub_data})